REGISTRY = {}

from .rnn_agent import RNNAgent
REGISTRY["rnn"] = RNNAgent

from .multi_head_agent import MultiHeadAgent
REGISTRY["multi_head"] = MultiHeadAgent